from .cifar import get_cifar10
from .fixmatch_cifar import get_cifar100_fixmatch,get_cifar10_fixmatch

def build_loader(args):
    if args.datasets =="cifar10":
        labeled_loader, unlabeled_loader, test_loader =get_cifar10(
            root=args.data_path,
            label_num= args.label_num,
            batch_size= args.batch_size,
            unlabel_batch_size= args.unlabel_batch_size
        )
        return labeled_loader, unlabeled_loader, test_loader
    elif args.datasets == "fixmatch_cifar10":
        labeled_loader, unlabeled_loader, test_loader =get_cifar10_fixmatch(
            root=args.data_path,
            label_num= args.label_num,
            batch_size= args.batch_size,
            unlabel_batch_size= args.unlabel_batch_size
        )
        return labeled_loader, unlabeled_loader, test_loader
    
    
